from Config import db
from model.note import Note


def addNote(title, body, user_id):
    '''
    添加文章
    :param title:文章的标题
    :param body:文章的主体
    :param user_id:文章的作者id
    :return:None
    '''
    note = Note(title, body, user_id)
    db.session.add(note)
    db.session.commit()


def upNote(note_id, title, body):
    '''
    更新当前文章的内容或者标题
    :param note_id: 当前文章的id
    :return:None
    '''
    note = Note.query.get(note_id)
    note.title = title
    note.body = body
    db.session.add(note)
    db.session.commit()


def delete_note(id):
    '''
    通过文章id删除文章
    :param id:文章id
    :return:None
    '''
    note = Note.query.get(id)
    db.session.delete(note)
    db.session.commit()


def query_note(current_user):
    '''
    显示出当前用户所有的日记标题和文章主主体
    :param current_user: 当前用户的id
    :return:List<note>
    '''
    note_list = Note.query.filter_by(user_id=current_user).all()
    return note_list
